/*
 * Copyright 2002 Sun Microsystems, Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 *
 * - Redistribution in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in
 *   the documentation and/or other materials provided with the
 *   distribution.
 *
 * Neither the name of Sun Microsystems, Inc. or the names of
 * contributors may be used to endorse or promote products derived
 * from this software without specific prior written permission.
 *
 * This software is provided "AS IS," without a warranty of any
 * kind. ALL EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND
 * WARRANTIES, INCLUDING ANY IMPLIED WARRANTY OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE HEREBY
 * EXCLUDED. SUN AND ITS LICENSORS SHALL NOT BE LIABLE FOR ANY DAMAGES
 * SUFFERED BY LICENSEE AS A RESULT OF USING, MODIFYING OR
 * DISTRIBUTING THE SOFTWARE OR ITS DERIVATIVES. IN NO EVENT WILL SUN
 * OR ITS LICENSORS BE LIABLE FOR ANY LOST REVENUE, PROFIT OR DATA, OR
 * FOR DIRECT, INDIRECT, SPECIAL, CONSEQUENTIAL, INCIDENTAL OR
 * PUNITIVE DAMAGES, HOWEVER CAUSED AND REGARDLESS OF THE THEORY OF
 * LIABILITY, ARISING OUT OF THE USE OF OR INABILITY TO USE SOFTWARE,
 * EVEN IF SUN HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
 *
 * You acknowledge that Software is not designed, licensed or intended
 * for use in the design, construction, operation or maintenance of
 * any nuclear facility.
 */

package com.sun.j2ee.blueprints.supplierpo.ejb;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.StringReader;
import java.net.URL;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.Iterator;

import javax.xml.parsers.SAXParserFactory;
import javax.xml.transform.Result;
import javax.xml.transform.Source;
import javax.xml.transform.sax.SAXSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;

import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.XMLFilter;
import org.xml.sax.XMLReader;
import org.xml.sax.helpers.XMLFilterImpl;

import com.sun.j2ee.blueprints.contactinfo.ejb.ContactInfo;
import com.sun.j2ee.blueprints.lineitem.ejb.LineItem;
import com.sun.j2ee.blueprints.xmldocuments.XMLDocumentException;
import com.sun.j2ee.blueprints.xmldocuments.XMLDocumentUtils;

public class SupplierOrder {
	public static final boolean TRACE = true;
	public static final String DTD_PUBLIC_ID = "-//Sun Microsystems, Inc. - J2EE Blueprints Group//DTD SupplierOrder 1.1//EN";
	public static final String DTD_SYSTEM_ID = "/com/sun/j2ee/blueprints/supplierpo/rsrc/schemas/SupplierOrder.dtd";
	public static final boolean VALIDATING = true;
	public static final String XML_SUPPLIERORDER = "SupplierOrder";
	public static final String XML_ORDERID = "OrderId";
	public static final String XML_ORDERDATE = "OrderDate";
	public static final String XML_SHIPPINGINFO = "ShippingInfo";
	private final SimpleDateFormat dateFormat = new SimpleDateFormat(
			"yyyy-MM-dd");
	private String orderId;
	private Date orderDate;
	private ContactInfo shippingInfo;
	private ArrayList<LineItem> lineItems = null;

	// Constructor to be used when creating SupplierOrder from data

	public SupplierOrder() {
	}

	// getter methods

	public String getOrderId() {
		return orderId;
	}

	public Date getOrderDate() {
		return orderDate;
	}

	public ContactInfo getShippingInfo() {
		return shippingInfo;
	}

	public Collection<LineItem> getLineItems() {
		return lineItems;
	}

	// setter methods

	public void setOrderId(String orderId) {
		this.orderId = orderId;
	}

	public void setOrderDate(Date orderDate) {
		this.orderDate = orderDate;
	}

	public void setShippingInfo(ContactInfo contactInfo) {
		this.shippingInfo = contactInfo;
		return;
	}

	public void addLineItem(LineItem lineItem) {
		if (lineItems == null) {
			lineItems = new ArrayList<LineItem>();
		}
		lineItems.add(lineItem);
		return;
	}

	// XML (de)serialization methods

	public void toXML(Result result) throws XMLDocumentException {
		toXML(result, null);
		return;
	}

	public String toXML() throws XMLDocumentException {
		return toXML((URL) null);
	}

	public void toXML(Result result, URL entityCatalogURL)
			throws XMLDocumentException {
		if (entityCatalogURL != null) {
			XMLDocumentUtils.toXML(toDOM(), DTD_PUBLIC_ID, entityCatalogURL,
					XMLDocumentUtils.DEFAULT_ENCODING, result);
		} else {
			XMLDocumentUtils.toXML(toDOM(), DTD_PUBLIC_ID, DTD_SYSTEM_ID,
					XMLDocumentUtils.DEFAULT_ENCODING, result);
		}
		return;
	}

	public String toXML(URL entityCatalogURL) throws XMLDocumentException {
		try {
			ByteArrayOutputStream stream = new ByteArrayOutputStream();
			toXML(new StreamResult(stream), entityCatalogURL);
			if (TRACE) {
				System.err.println("SupplierOrder.toXML: "
						+ stream.toString(XMLDocumentUtils.DEFAULT_ENCODING));
			}
			return stream.toString(XMLDocumentUtils.DEFAULT_ENCODING);
		} catch (Exception exception) {
			throw new XMLDocumentException(exception);
		}
	}

	public static SupplierOrder fromXML(Source source)
			throws XMLDocumentException {
		return fromXML(source, null, VALIDATING);
	}

	public static SupplierOrder fromXML(String buffer)
			throws XMLDocumentException {
		return fromXML(buffer, null, VALIDATING);
	}

	public static SupplierOrder fromXML(Source source, URL entityCatalogURL,
			boolean validating) throws XMLDocumentException {
		return fromDOM(XMLDocumentUtils.fromXML(source, DTD_PUBLIC_ID,
				entityCatalogURL, validating));
	}

	public static SupplierOrder fromXML(String buffer, URL entityCatalogURL,
			boolean validating) throws XMLDocumentException {
		System.err.println(buffer);
		try {
			return fromXML(new StreamSource(new StringReader(buffer)),
					entityCatalogURL, validating);
		} catch (XMLDocumentException exception) {
			System.err.println(exception.getRootCause().getMessage());
			throw new XMLDocumentException(exception);
		}
	}

	public Document toDOM() throws XMLDocumentException {
		Document document = XMLDocumentUtils.createDocument();
		Element root = (Element) toDOM(document);
		document.appendChild(root);
		return document;
	}

	public Node toDOM(Document document) {
		Element root = document.createElement(XML_SUPPLIERORDER);
		XMLDocumentUtils.appendChild(document, root, XML_ORDERID, orderId);
		XMLDocumentUtils.appendChild(document, root, XML_ORDERDATE, dateFormat
				.format(orderDate));
		Element element = document.createElement(XML_SHIPPINGINFO);
		element.appendChild(shippingInfo.toDOM(document));
		root.appendChild(element);
		for (Iterator<LineItem> i = lineItems.iterator(); i.hasNext();) {
			LineItem lineItem = i.next();
			root.appendChild(lineItem.toDOM(document));
		}
		return root;
	}

	public static SupplierOrder fromDOM(Document document)
			throws XMLDocumentException {
		return fromDOM(document.getDocumentElement());
	}

	public static SupplierOrder fromDOM(Node node) throws XMLDocumentException {
		Element element;
		if (node.getNodeType() == Node.ELEMENT_NODE
				&& (element = ((Element) node)).getTagName().equals(
						XML_SUPPLIERORDER)) {
			Element child;
			SupplierOrder supplierOrder = new SupplierOrder();
			child = XMLDocumentUtils.getFirstChild(element, XML_ORDERID, false);
			supplierOrder.setOrderId(XMLDocumentUtils.getContentAsString(child,
					false));
			try {
				child = XMLDocumentUtils.getNextSibling(child, XML_ORDERDATE,
						false);
				supplierOrder.setOrderDate(supplierOrder.dateFormat
						.parse(XMLDocumentUtils
								.getContentAsString(child, false)));
			} catch (Exception exception) {
				supplierOrder.setOrderDate(new Date()); // FIX ME!
				System.err.println(XML_ORDERDATE + ": "
						+ exception.getMessage() + " reset to current date.");
			}
			child = XMLDocumentUtils.getNextSibling(child, XML_SHIPPINGINFO,
					false);
			supplierOrder.shippingInfo = ContactInfo.fromDOM(XMLDocumentUtils
					.getFirstChild(child, ContactInfo.XML_CONTACTINFO, false));
			for (child = XMLDocumentUtils.getNextSibling(child,
					LineItem.XML_LINEITEM, false); child != null; child = XMLDocumentUtils
					.getNextSibling(child, LineItem.XML_LINEITEM, true)) {
				supplierOrder.addLineItem(LineItem.fromDOM(child));
			}
			return supplierOrder;
		}
		throw new XMLDocumentException(XML_SUPPLIERORDER + " element expected.");
	}

	public static void main(String[] args) {
		if (args.length <= 1) {
			String fileName = args.length > 0 ? args[0] : "SupplierOrder.xml";
			try {
				SupplierOrder supplierOrder = SupplierOrder
						.fromXML(new StreamSource(new FileInputStream(new File(
								fileName)), fileName));
				supplierOrder.toXML(new StreamResult(System.out));
				SAXParserFactory parserFactory = SAXParserFactory.newInstance();
				parserFactory.setValidating(true);
				XMLReader reader = parserFactory.newSAXParser().getXMLReader();
				XMLFilter filter = new XMLFilterImpl(reader) {
					public void startDocument() throws SAXException {
						System.err.println("StartDocument");
						getContentHandler().startDocument();
						return;
					}
				};
				SAXSource saxSource = new SAXSource(filter, new InputSource(
						fileName));
				supplierOrder = SupplierOrder.fromXML(saxSource);
				supplierOrder.toXML(new StreamResult(System.out));
				System.exit(0);
			} catch (IOException exception) {
				exception.printStackTrace(System.err);
				System.err.println(exception);
				System.exit(2);
			} catch (XMLDocumentException exception) {
				exception.printStackTrace(System.err);
				System.err.println(exception.getRootCause());
				System.exit(2);
			} catch (Exception exception) {
				exception.printStackTrace(System.err);
				System.err.println(exception);
				System.exit(2);
			}
		}
		System.err.println("Usage: " + SupplierOrder.class.getName()
				+ " [file-name]");
		System.exit(1);
	}
}
